package homework;

import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;

public class Server {
    private ServerSocket serverSocket;
    //private PrintWriter[] allOut = {};
    private Collection<PrintWriter> allOut=new ArrayList<>();

    public Server() {
        try {
            System.out.println("正在启动服务器");
            serverSocket = new ServerSocket(8088);
            System.out.println("启动服务器成功");
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public void start() {

        try {
            while (true) {
                System.out.println("等待客户端连接");
                Socket socket = serverSocket.accept();
                System.out.println("客户端连接成功");
                Runnable handler = new ClientHandler(socket);
                Thread t = new Thread(handler);
                t.start();


            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) {
        Server server = new Server();
        server.start();
    }

    public class ClientHandler implements Runnable {
        private Socket socket;
        private String host;

        public ClientHandler(Socket socket) {
            this.socket = socket;
            host = socket.getInetAddress().getHostAddress();

        }

        public void run() {
            PrintWriter pw = null;
            try {
                InputStream in = socket.getInputStream();
                InputStreamReader isr = new InputStreamReader(in, StandardCharsets.UTF_8);
                BufferedReader br = new BufferedReader(isr);

                OutputStream out = socket.getOutputStream();
                OutputStreamWriter osw = new OutputStreamWriter(out, StandardCharsets.UTF_8);
                BufferedWriter bw = new BufferedWriter(osw);
                pw = new PrintWriter(bw, true);

                synchronized (Server.this) {
                   allOut.add(pw);
                }
                sendMassage(host + "上线了，当前在线人数：" + allOut.size());
                String line;
                while ((line = br.readLine()) != null) {
                    sendMassage(host + "说：" + line);
                }
            } catch (IOException e) {
                e.printStackTrace();
            } finally {

                synchronized (Server.this) {
                   allOut.remove(pw);
                }
            }
            sendMassage(host + "下线了，当前在线人数：" + allOut.size());
            try {
                socket.close();
            } catch (IOException e) {
                e.printStackTrace();
            }

        }

        public void sendMassage(String line) {
            synchronized (Server.this) {
                for (PrintWriter pw:allOut) {
                   pw.println(line);
                }
            }
        }
    }
}
